hacker rank mathematics solution with python